import cv2
from easyocr import Reader

# 1、初始化Reader 对象，用来标记语言
# 第一次加载会自动下载文字识别的模板
reader = Reader(['en'])

# 2、读取图片
img_path = '../data/ocrdata/yingwen1.jpeg'
img = cv2.imread(img_path)
# 3、调用easyocr 识别文本
result = reader.readtext(img_path)
print(result)
# 文本存储
full_text = ''
# 输出识别 结果
print("开始识别结果：")
for bbox, text, confidence in result:
    if confidence > 0.6:
        # 置信度大于0.7 进行文本读取
        print(f'文本：{text}，置信度：{confidence:.2f}')
        full_text += text + "\n"
        print(f"仅保留字母{bbox[0]},{bbox[1]},{full_text}")
    else:
        print(f"已过滤低的置信度：{text}，置信度：{confidence:.2f}")

# 去除末尾多余的空格
full_text = full_text.strip()
print("处理之后的高置信度文本：", full_text)

cv2.imshow('img', img)
cv2.waitKey(0)
cv2.destroyAllWindows()
